from face_matching import FaceMatching
import os
import json
# 初始化人脸匹配类
def main():
    # 创建FaceMatching实例
    face_matcher = FaceMatching(similarity_threshold=0.5)
    
    # 源图片和目标图片路径
    src_dir = 'static/images/src-face/chenxun'
    tar_dir = 'static/images/tar-face/chenxun'
    
    # 确保目录存在
    if not os.path.exists(src_dir):
        print(f"源图片目录不存在: {src_dir}")
        return
    if not os.path.exists(tar_dir):
        print(f"目标图片目录不存在: {tar_dir}")
        return
    
    # 获取源目录中的所有图片
    src_files = [f for f in os.listdir(src_dir) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
    
    if not src_files:
        print("源目录中没有图片")
        return
    
    # 示例1: 匹配两张图片
    if len(src_files) >= 2:
        img1 = os.path.join(src_dir, src_files[0])
        img2 = os.path.join(src_dir, src_files[1])
        result = face_matcher.match_two_faces(img1, img2)
        print("\n两张图片匹配结果:")
        print(json.dumps(result, indent=2, ensure_ascii=False))
    
    # 示例2: 匹配源图片与目标目录
    if src_files:
        source_img = os.path.join(src_dir, src_files[0])
        result = face_matcher.match_face_to_targets(source_img, tar_dir)
        print("\n源图片与目标目录匹配结果:")
        print(json.dumps(result, indent=2, ensure_ascii=False))

if __name__ == "__main__":
    main()